// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

import { DatasetTaskType, IDataset } from "@responsible-ai/core-ui";

import { images } from "./images";

export const imagenet: IDataset = {
  categorical_features: [],
  class_names: [
    "tench",
    "goldfish",
    "great_white_shark",
    "tiger_shark",
    "hammerhead",
    "electric_ray",
    "stingray",
    "cock",
    "hen",
    "ostrich",
    "brambling",
    "goldfinch",
    "house_finch",
    "junco",
    "indigo_bunting",
    "robin",
    "bulbul",
    "jay",
    "magpie",
    "chickadee",
    "water_ouzel",
    "kite",
    "bald_eagle",
    "vulture",
    "great_grey_owl",
    "European_fire_salamander",
    "common_newt",
    "eft",
    "spotted_salamander",
    "axolotl",
    "bullfrog",
    "tree_frog",
    "tailed_frog",
    "loggerhead",
    "leatherback_turtle",
    "mud_turtle",
    "terrapin",
    "box_turtle",
    "banded_gecko",
    "common_iguana",
    "American_chameleon",
    "whiptail",
    "agama",
    "frilled_lizard",
    "alligator_lizard",
    "Gila_monster",
    "green_lizard",
    "African_chameleon",
    "Komodo_dragon",
    "African_crocodile",
    "American_alligator",
    "triceratops",
    "thunder_snake",
    "ringneck_snake",
    "hognose_snake",
    "green_snake",
    "king_snake",
    "garter_snake",
    "water_snake",
    "vine_snake",
    "night_snake",
    "boa_constrictor",
    "rock_python",
    "Indian_cobra",
    "green_mamba",
    "sea_snake",
    "horned_viper",
    "diamondback",
    "sidewinder",
    "trilobite",
    "harvestman",
    "scorpion",
    "black_and_gold_garden_spider",
    "barn_spider",
    "garden_spider",
    "black_widow",
    "tarantula",
    "wolf_spider",
    "tick",
    "centipede",
    "black_grouse",
    "ptarmigan",
    "ruffed_grouse",
    "prairie_chicken",
    "peacock",
    "quail",
    "partridge",
    "African_grey",
    "macaw",
    "sulphur-crested_cockatoo",
    "lorikeet",
    "coucal",
    "bee_eater",
    "hornbill",
    "hummingbird",
    "jacamar",
    "toucan",
    "drake",
    "red-breasted_merganser"
  ],
  feature_names: ["mean_pixel_value"],
  features: [
    [113.15779113769531],
    [91.82549285888672],
    [86.30901336669922],
    [157.78981018066406],
    [136.93569946289062],
    [115.3836898803711],
    [102.74627685546875],
    [80.134521484375],
    [76.43243408203125],
    [135.4658203125],
    [58.03116989135742],
    [85.90875244140625],
    [106.01287841796875],
    [116.58745574951172],
    [84.64251708984375],
    [59.607486724853516],
    [109.2101058959961],
    [122.94914245605469],
    [147.7889862060547],
    [169.5949249267578],
    [97.15763854980469],
    [163.70245361328125],
    [176.89952087402344],
    [124.58245849609375],
    [127.13233184814453],
    [88.27935791015625],
    [95.46903228759766],
    [136.11721801757812],
    [143.87786865234375],
    [242.28895568847656],
    [101.06806945800781],
    [30.1624813079834],
    [78.04510498046875],
    [172.19473266601562],
    [158.976318359375],
    [95.03946685791016],
    [51.86744689941406],
    [121.98445129394531],
    [85.8304672241211],
    [110.73126983642578],
    [215.4990234375],
    [97.17032623291016],
    [176.9586181640625],
    [141.5923614501953],
    [79.47926330566406],
    [107.56243896484375],
    [90.36686706542969],
    [140.5702362060547],
    [194.81246948242188],
    [183.98062133789062]
  ],
  images,
  predicted_y: [
    90, 13, 81, 74, 42, 74, 73, 55, 83, 97, 80, 13, 91, 97, 34, 10, 76, 35, 72,
    89, 10, 45, 54, 49, 32, 45, 20, 31, 88, 87, 39, 90, 78, 61, 90, 24, 61, 61,
    57, 42, 77, 29, 62, 52, 30, 97, 57, 40, 9, 95
  ],
  target_column: "label",
  task_type: DatasetTaskType.ImageClassification,
  true_y: [
    90, 13, 81, 74, 42, 74, 73, 55, 83, 97, 80, 13, 91, 97, 34, 10, 76, 35, 72,
    89, 10, 45, 54, 4, 3, 5, 20, 71, 87, 77, 12, 48, 85, 68, 79, 15, 35, 26, 18,
    27, 84, 93, 58, 34, 62, 34, 47, 96, 13, 68
  ]
};
